package com.ithui.tidal.websocket;

import cn.hutool.extra.spring.SpringUtil;
import cn.hutool.json.JSONUtil;
import com.ithui.tidal.websocket.domain.enums.WSReqTypeEnum;
import com.ithui.tidal.websocket.domain.service.WebSocketService;
import com.ithui.tidal.websocket.domain.vo.req.WSBaseReq;
import io.netty.channel.Channel;
import io.netty.channel.ChannelHandler.Sharable;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler;
import io.netty.handler.timeout.IdleState;
import io.netty.handler.timeout.IdleStateEvent;
import lombok.extern.slf4j.Slf4j;

/**
 * @Author zn5267
 **/
@Slf4j
@Sharable // 标注为可共享的,不加此注解会报错
public class NettyWebSocketServerHandler extends SimpleChannelInboundHandler<TextWebSocketFrame> {

    private WebSocketService webSocketService;

    /**
     * 这里的构造方法是为了注入WebSocketService实例
     *
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        webSocketService = SpringUtil.getBean(WebSocketService.class);
        // 连接保存到WebSocketService中
        webSocketService.connect(ctx.channel());
    }

    /**
     * 客户端断开连接时调用
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        userOffline(ctx.channel());
    }

    /**
     * 处理websocket连接建立，握手事件
     * 第一次握手成功后，netty会自动调用该方法，将handler添加到pipeline中
     * 第二次会调用handlerAdded方法，将handler从pipeline中移除
     *
     * @param ctx
     * @param evt
     * @throws Exception
     */
    @Override
    public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception {
        // 握手完成事件
        if (evt instanceof WebSocketServerProtocolHandler.HandshakeComplete) {
            System.out.println("握手完成");
        } else if (evt instanceof IdleStateEvent) { // 心跳包
            IdleStateEvent event = (IdleStateEvent) evt;
            // 如果是读空闲状态
            if (event.state().equals(IdleState.READER_IDLE)) {
                System.out.println("读空闲状态");
                // todo 用户下线
            }

        }
    }

    /**
     * 用户下线统一处理
     * @param channel
     */
    private void userOffline(Channel channel) {
        webSocketService.remove(channel);
        channel.close();
    }


    /**
     * 处理websocket消息
     * 客户端发送消息到服务端，netty会自动调用该方法，将消息传递给handler进行处理
     *
     * @param ctx
     * @param msg
     * @throws Exception
     */
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, TextWebSocketFrame msg) throws Exception {
        String text = msg.text();
        System.out.println("Received message: " + text);
        // 通过json转换为对象
        WSBaseReq wsBaseReq = JSONUtil.toBean(text, WSBaseReq.class);
        // 根据type进行不同的业务处理
        switch (WSReqTypeEnum.of(wsBaseReq.getType())) {
            case LOGIN:
                // 登录业务处理,微信登录
                webSocketService.handleLoginReq(ctx.channel());
              /*  System.out.println("微信请求二维码");
                // 返回数据的时候不能直接返回对象，需要返回TextWebSocketFrame对象
                ctx.channel().writeAndFlush(new TextWebSocketFrame("123"));*/
                break;
            case HEARTBEAT:
                // 心跳包业务处理
                break;
            case AUTHORIZE:
                // 登录认证业务处理
                break;
        }

    }
}
